package chapter01;

import java.util.HashMap;

/**
 * 两数之和
 * 给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出和为目标值 target的那两个整数，并返回它们的数组下标。
 * 你可以假设每种输入只会对应一个答案，并且你不能使用两次相同的元素。
 * 你可以按任意顺序返回答案。
 * <p>
 * 输入：nums = [2,7,11,15], target = 9
 * 输出：[0,1]
 * 解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。
 */
public class TwoSum {
    // 暴力法
    public static int[] twoSum(int[] nums, int target) {
        int[] res = new int[2];
        for (int i = 0; i < nums.length; i++) {
            for (int j = i + 1; j < nums.length; j++) {
                if (nums[i] + nums[j] == target) {
                    res[0] = i;
                    res[1] = j;
                }
            }
        }
        return res;
    }

    /**
     * 哈希表法
     * 在上面暴力法中，寻找 target-nums[i]最耗时。
     * 如果对于每一个 nums[i]，我们首先查询哈希表中是否存在 target - nums[i]
     * 然后将 nums[i] 插入到哈希表中，即可保证不会让 nums[i]和自己匹配。
     */
    public static int[] twoSum1(int[] nums, int target) {
        int[] res = new int[2];
        HashMap<Integer, Integer> hashtable = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            if (hashtable.containsKey(target - nums[i])) {
                res[0] = i;
                res[1] = hashtable.get(target - nums[i]);
                return res;
            }
            hashtable.put(nums[i], i);
        }
        return res;
    }
}
